# 从文件中读取并反序列化对象
import json
import os
import pickle

from ai_btn_mate import get_ai_btn_mate
from get_btn import reset_device_btn_content_dict, get_btn


def test_get_btn():
    # reset_device_btn_content_dict()

    print(get_btn("xiaomi2"))

    # 获取当前脚本的文件路径
    current_file_path = os.path.abspath(__file__)

    # 获取当前脚本所在的文件夹路径
    current_directory = os.path.dirname(current_file_path)

    # 从文件中读取并反序列化对象
    with open(current_directory + '/device_btn_content_dict.pkl', 'rb') as file:
        device_btn_content_dict = pickle.load(file)

    print(device_btn_content_dict)

    # 查看本地是否有该设备的数据
    # btn_content_dict = device_btn_content_dict.get("xiaomi2")

    # print(btn_content_dict)

    # print(upset_btn_content_dict(get_btn_content_dict()))
    #
    # print(get_btn_content_dict())


def test_get_ai_btn_mate():
    print(get_ai_btn_mate("最近是挺热的"))



if __name__ == "__main__":

    # test_get_btn()
    test_get_ai_btn_mate()


